# coding: utf-8
import torch


class ProjectConfig(object):
    def __init__(self):
        self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
        self.pre_model = 'D:\\Project\\AIStudent\\LLMProject\\BERT_Fintuning\\PET\\bert-base-chinese'
        self.train_path = 'D:\\Project\\AIStudent\\LLMProject\\BERT_Fintuning\\PET\\data\\train.txt'
        self.dev_path = 'D:\\Project\\AIStudent\\LLMProject\\BERT_Fintuning\\PET\\data\\dev.txt'
        self.prompt_file = 'D:\\Project\\AIStudent\\LLMProject\\BERT_Fintuning\\PET\\data\\prompt.txt'
        self.verbalizer = 'D:\\Project\\AIStudent\\LLMProject\\BERT_Fintuning\PET\\data\\verbalizer.txt'
        self.max_len = 512
        self.batch_size = 8
        self.learning_rate = 5e-5
        self.weight_decay = 0
        self.warmup_ratio = 0.06
        self.max_label_len = 2
        self.epochs = 50
        self.logging_step = 1
        self.valid_steps = 10
        self.save_dir = 'D:\\Project\\AIStudent\\LLMProject\\BERT_Fintuning\\PET\\checkpoints'

if __name__ == '__main__':
    pc = ProjectConfig()
    print(pc.prompt_file)
    print(pc.pre_model)